package com.dgut.lab.service;

import com.dgut.lab.entity.Questionnaire;
import com.dgut.lab.mapper.QuestionnaireRepository;
import org.springframework.data.domain.Example;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class QuestionnaireService  {

    @Resource
    private QuestionnaireRepository questionnaireRepository;
    @Resource
    private MongoTemplate mongoTemplate;

    public Questionnaire addQuestionnaire(Questionnaire questionnaire) {
        return questionnaireRepository.insert(questionnaire);
    }

    public Questionnaire queryById(Questionnaire questionnaire) {
        return questionnaireRepository.findById(questionnaire.getQuestionnaireId()).get();
    }

    public Questionnaire queryLatest() {
        Questionnaire questionnaire = new Questionnaire();
        questionnaire.setEnable(1);
        Example<Questionnaire> example = Example.of(questionnaire);
        Sort sort = Sort.by(Sort.Direction.DESC, "_id");
        PageRequest pageRequest = PageRequest.of(0,1,sort);
        List<Questionnaire> results =  questionnaireRepository.findAll(example,pageRequest).getContent();
        if(results.get(0)!=null){
            return results.get(0);
        }
        return null;
    }
}
